package servlets;

import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Iterator;
import java.util.List;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.DiskFileUpload;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;

import services.GeneralFunctions;

/**
 * Класс реализации сервлета UploadImageServlet, который служит для загрузки
 * изображений с компьютера пользователя на сервер.
 */
@WebServlet("/UploadImageServlet")
public class UploadImageServlet extends HttpServlet {

	private static final long serialVersionUID = 1L;

	public UploadImageServlet() { }

	/**
	 * Собственно, метод, который вызывается при загрузке изображения на сервер.
	 */
	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		DiskFileUpload upload = new DiskFileUpload();
		List<?> items = null;
		try {
			items = upload.parseRequest(request);
		} catch (FileUploadException e) {
			e.printStackTrace();
		}

		Iterator<?> iterator = items.iterator();
		FileItem item = null;
		while (iterator.hasNext()) {
			item = (FileItem) iterator.next();
			if (!item.isFormField()) break;
		}
		byte[] img = new byte[(int) item.getSize()];
		item.getInputStream().read(img);

		InputStream in = new ByteArrayInputStream(img);
		BufferedImage image = ImageIO.read(in);

		// Сохраняем загруженное изображение под новым именем.
		String name = GeneralFunctions.generateName();
		ImageIO.write(image, "png", new File(GeneralFunctions.getPathToImages() + name + ".png"));

		// Удаляем предыдущее изображение, находившееся в обработке.
		if (request.getSession().getAttribute("processing_seamless_cloning") == null) {
			String oldImageName = (String) request.getSession().getAttribute("current_image");
			if (!oldImageName.equals("")) GeneralFunctions.deleteImage(oldImageName);
		}

		// Устанавливаем загруженное изображение текущим и перенаправляем запрос на index.jsp.
		request.getSession().setAttribute("current_image", name);
		if (request.getSession().getAttribute("processing_seamless_cloning") != null) {
			request.getRequestDispatcher("/index.jsp?text=seamless_cloning&subtext=loaded_photo2").forward(request, response);
		} else
		request.getRequestDispatcher("/index.jsp").forward(request, response);
	}

}
